how to make python code faster